package BinarySortTree;

/*
二叉搜索树中的插入操作
给定二叉搜索树（BST）的根节点root和要插入树中的值value，将值插入二叉搜索树。
返回插入后二叉搜索树的根节点。 输入数据 保证 ，新值和原始二叉搜索树中的任意节点值都不同。
注意，可能存在多种有效的插入方式，只要树在插入后仍保持为二叉搜索树即可。 你可以返回 任意有效的结果 。

示例 1：
输入：root = [4,2,7,1,3], val = 5
输出：[4,2,7,1,3,5]
解释：另一个满足题目要求可以通过的树是：
示例 2：
输入：root = [40,20,60,10,30,50,70], val = 25
输出：[40,20,60,10,30,50,70,null,null,25]
示例 3：
输入：root = [4,2,7,1,3,null,null,null,null,null,null], val = 5
输出：[4,2,7,1,3,5]

作者：LeetCode
链接：https://leetcode.cn/leetbook/read/introduction-to-data-structure-binary-search-tree/xp1llt/
 */

public class _22二叉搜索树中的插入操作 {
    public static void main(String[] args) {

    }

    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }

    //递归
    public TreeNode insertIntoBST(TreeNode root, int val) {
        return dfs(root,val,root);
    }
    public TreeNode dfs(TreeNode root, int val,TreeNode initroot) {
        if (root == null) {
            initroot = new TreeNode(val);
            return initroot;
        }
        if(root.val > val){
            if(root.left == null){
                root.left = new TreeNode(val);
                return initroot;
            }
            return dfs(root.left, val, initroot);
        }else {
            if(root.right == null){
                root.right = new TreeNode(val);
                return initroot;
            }
            return dfs(root.right, val, initroot);
        }
    }

    //官解：迭代
    class Solution {
        public TreeNode insertIntoBST(TreeNode root, int val) {
            if (root == null) {
                return new TreeNode(val);
            }
            TreeNode pos = root;
            while (pos != null) {
                if (val < pos.val) {
                    if (pos.left == null) {
                        pos.left = new TreeNode(val);
                        break;
                    } else {
                        pos = pos.left;
                    }
                } else {
                    if (pos.right == null) {
                        pos.right = new TreeNode(val);
                        break;
                    } else {
                        pos = pos.right;
                    }
                }
            }
            return root;
        }
    }


}
